

package com.hazelcast.spi.impl.eventservice.impl.operations;

import com.hazelcast.internal.util.UUIDSerializationUtil;
import com.hazelcast.nio.ObjectDataInput;
import com.hazelcast.nio.ObjectDataOutput;
import com.hazelcast.spi.impl.SpiDataSerializerHook;
import com.hazelcast.spi.impl.eventservice.impl.EventServiceImpl;
import com.hazelcast.spi.impl.eventservice.impl.EventServiceSegment;

import java.io.IOException;
import java.util.UUID;

public class DeregistrationOperation extends AbstractRegistrationOperation {

    private String topic;
    private UUID id;

    public DeregistrationOperation() {
    }

    public DeregistrationOperation(String topic, UUID id, int orderKey, int memberListVersion) {
        super(memberListVersion, orderKey);
        this.topic = topic;
        this.id = id;
    }

    @Override
    protected void runInternal() {
        EventServiceImpl eventService = (EventServiceImpl) getNodeEngine().getEventService();
        EventServiceSegment segment = eventService.getSegment(getServiceName(), false);
        if (segment != null) {
            if (id == null) {
                segment.removeRegistrations(topic);
            } else {
                segment.removeRegistration(topic, id);
            }
        }
    }

    @Override
    protected void writeInternalImpl(ObjectDataOutput out) throws IOException {
        out.writeString(topic);
        UUIDSerializationUtil.writeUUID(out, id);
    }

    @Override
    protected void readInternalImpl(ObjectDataInput in) throws IOException {
        topic = in.readString();
        id = UUIDSerializationUtil.readUUID(in);
    }

    @Override
    public int getClassId() {
        return SpiDataSerializerHook.DEREGISTRATION;
    }
}
